// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "ui/shell_dialogs/base_shell_dialog_win.h"

#include <algorithm>

#include "base/threading/thread.h"
#include "base/win/scoped_com_initializer.h"

namespace ui {

// static
BaseShellDialogImpl::Owners BaseShellDialogImpl::owners_;
int BaseShellDialogImpl::instance_count_ = 0;

BaseShellDialogImpl::BaseShellDialogImpl()
{
    ++instance_count_;
}

BaseShellDialogImpl::~BaseShellDialogImpl()
{
    // All runs should be complete by the time this is called!
    if (--instance_count_ == 0)
        DCHECK(owners_.empty());
}

BaseShellDialogImpl::RunState BaseShellDialogImpl::BeginRun(HWND owner)
{
    // Cannot run a modal shell dialog if one is already running for this owner.
    DCHECK(!IsRunningDialogForOwner(owner));
    // The owner must be a top level window, otherwise we could end up with two
    // entries in our map for the same top level window.
    DCHECK(!owner || owner == GetAncestor(owner, GA_ROOT));
    RunState run_state;
    run_state.dialog_thread = CreateDialogThread();
    run_state.owner = owner;
    if (owner) {
        owners_.insert(owner);
        DisableOwner(owner);
    }
    return run_state;
}

void BaseShellDialogImpl::EndRun(RunState run_state)
{
    if (run_state.owner) {
        DCHECK(IsRunningDialogForOwner(run_state.owner));
        EnableOwner(run_state.owner);
        DCHECK(owners_.find(run_state.owner) != owners_.end());
        owners_.erase(run_state.owner);
    }
    DCHECK(run_state.dialog_thread);
    delete run_state.dialog_thread;
}

bool BaseShellDialogImpl::IsRunningDialogForOwner(HWND owner) const
{
    return (owner && owners_.find(owner) != owners_.end());
}

void BaseShellDialogImpl::DisableOwner(HWND owner)
{
    if (IsWindow(owner))
        EnableWindow(owner, FALSE);
}

// static
base::Thread* BaseShellDialogImpl::CreateDialogThread()
{
    base::Thread* thread = new base::Thread("Chrome_ShellDialogThread");
    thread->init_com_with_mta(false);
    bool started = thread->Start();
    DCHECK(started);
    return thread;
}

void BaseShellDialogImpl::EnableOwner(HWND owner)
{
    if (IsWindow(owner))
        EnableWindow(owner, TRUE);
}

} // namespace ui
